#include "UdpServer.hpp"
#include <memory>
#include <cstdio>
#include <vector>
#include "log.hpp"

using namespace std;

log Log;

void Usage(string proc)
{
    cout << "\n\tUsage: " << proc << " port[1024+]\n" << endl;
}


// string Handler1(const string& str)
// {
//     pid_t id = fork();
//     if(id == 0)
//     {
//         // ls -a -l  解析命令
//         // exec*();  进程程序替换接口去执行相应命令
//     }

//     // return res;
// }


bool SafeCheck(const string &cmd)
{
    vector<string> key_word = {
        "rm",
        "mv",
        "cp",
        "kill",
        "sudo",
        "unlink",
        "uninstall",
        "yum",
        "top"
    };

    for(auto &word : key_word)
    {
        auto pos = cmd.find(word);
        if(pos != string::npos)
        {
            return false;
        }
    }

    return true;
}


string ExcuteCommand(const string cmd)
{
    cout << "get a request cmd: " << cmd << endl;
    if(!SafeCheck(cmd)) return "Dangerous Cmd!";

    FILE *fp = popen(cmd.c_str(), "r"); // 不用自己解析字符串等等（创建管道、子进程、执行命令 并 返回 执行结果）
    if(fp == nullptr)
    {
        perror("popen");
        return "error";
    }

    string result;
    char buffer[1024];
    while(true)
    {
        char *res = fgets(buffer, sizeof(buffer), fp);
        if(res == nullptr) break;
        result += buffer;
    }

    pclose(fp);

    return result;
}

string Handler(const string& str, const string &clientip, uint16_t clientport)
{
    cout << "[" << clientip << ":" << clientport << "]#" << str << endl;
    string res = "Server get a string: ";
    res += str;
    cout << res << endl;
    return res;
}


// ./udpserver port
int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        Usage(argv[0]);
        exit(0);
    }

    uint16_t port = stoi(argv[1]);

    // unique_ptr<UdpServer> svr(new UdpServer(9000, "192.168.1.1"));
    unique_ptr<UdpServer> svr(new UdpServer(port));

    svr->Init();
    // svr->Run(ExcuteCommand);
    // svr->Run(Handler);
    svr->Run();

    return 0;
}